# -*- coding: utf-8 -*-
"""
获取日期时间的相关模块
"""
import datetime
from pandas import to_datetime


def today(has_line=0):
    '''获取今天的时间,返回20160701格式或者2016-07-01'''
    t=datetime.datetime.now()
    if has_line==0:  # 没有横线的格式
        return t.strftime('%Y%m%d')
    else:
        return t.strftime('%Y-%m-%d')

def which_week():
    '''返回今天是星期几。注意，星期天是6，星期1是0.需要转成我们平时习惯的'''
    t = datetime.datetime.now()
    return t.weekday()+1

def friday_date():
    '''获取本周周五的日期,因为可能是周6跑数，而不是周五'''
    t = datetime.datetime.now()
    t2 = t.weekday()+1  # 星期几
    if t2==5:
        return t.strftime('%Y%m%d')
    if t2 == 6:  # 如果是周6
        t = t + datetime.timedelta(-1)
        return t.strftime('%Y%m%d')
    if t2 == 7:  # 如果是周7
        t = t + datetime.timedelta(-2)
        return t.strftime('%Y%m%d')

def delta_days(date,num):
    '''计算date加减num天之后日期.    '''
    date = to_datetime(date)
    t = date+datetime.timedelta(num)
    t = t.strftime('%Y%m%d')
    return t
    